package com.castle.fortress.admin.argFlow.dto;

import lombok.Data;
import lombok.experimental.Accessors;
import java.util.Map;

/**
 * 数据集请求DTO
 */
@Data
@Accessors(chain = true)
public class DatasetRequestDTO {

    /**
     * 数据集名称
     */
    private String name;

    /**
     * 数据集头像
     */
    private String avatar;

    /**
     * 数据集描述
     */
    private String description;

    /**
     * 数据集语言
     */
    private String language;

    /**
     * 嵌入模型
     */
    private String embedding_model;

    /**
     * 权限设置
     */
    private String permission;

    /**
     * 分块方法 "naive"：通用
     *
     * "manual"：手动
     *
     * "qa"：问答
     *
     * "table"：表格
     *
     * "paper"：论文
     *
     * "book"：书籍
     *
     * "laws"：法律
     *
     * "presentation"：演示文稿
     *
     * "picture"：图片
     *
     * "one"：单个
     *
     * "knowledge_graph"：知识图谱
     * 在选择此选项之前，请确保您的 LLM 已在设置页面上正确配置。请注意，知识图谱会消耗大量 Tokens！
     *
     * "email"：电子邮件
     */
    private String chunk_method;

    /**
     * 解析器配置
     */
    private Map<String, Object> parser_config;

    /**
     * 向量模型 (已弃用，使用 embedding_model 替代)
     */
    @Deprecated
    private String vector_model;

    /**
     * 分组ID (已弃用)
     */
    @Deprecated
    private String group_id;
}
